import tempfile
import unittest
from pathlib import Path

from video_summarize_tools import edges
from video_summarize_tools.data_models import State


class EdgeTest(unittest.TestCase):
    def test_analysis_boot_edge(self):
        state = State()
        state["video_stream_file_path"] = None
        state["audio_stream_file_path"] = Path(tempfile.gettempdir())

        result = edges.analysis_boot_edge(state, None, None)
        self.assertEqual(result, "audio_transcript")
